(Menu) Clean up code