Better implementation of the menu thing..