Added main menu state and button class