Gov. Andrew Cuomo plans to visit states that have legalized marijuana to learn more about the issue and find out how New York could improve potential implementation. 

"I want to make sure it is done correctly," Cuomo said on Thursday. "You look at states that have legalized marijuana, each one of them have generated questions."

Cuomo said the tour could include Massachusetts, Illinois, Colorado and California.

The trips will include a discussion surrounding "what they've done, what's worked, has the social equity peice worked, as the law enforcement piece worked," Cuomo said at a news conference in New York City.  

The trip comes as lawmakers and the governor in the coming weeks will negotiate a finalized state budget that Cuomo has said should include the marijuana legalization provision. 

Cuomo in a radio interview on Wednesday called marijuana legalization a "top priority" for him in the state budget. 

State lawmakers in 2019 could not agree to a marijuana legalization measure and instead approved a back-stop bill that decriminalized possession and expunged criminal records. 

State elected officials, meanwhile, remain divided over how to approach how to spend marijuana revenue and how communities impacted by harsh drug laws should benefit from the law. 

The governor, meanwhile, has been working with neighboring states to develop a regional marijuana policy on tax rates and vaping usage.