I don't see why it should have any more restrictions on it than booze.
Make it where you have to be licensed to sell it, 21 to buy it, in trouble if you're caught smoking under 21 or buying for under 21, etc. Just like liquor.
Booze is a LOT more dangerous than pot, IMHO.
