Button and main menu
Code cleaning