Trying to implement main menu state