Are there any headshops in Mexico City?

I wish to move to Mexico City soon to join my friends, but I am concerned about the legality of such establishments. Please let me know...