Added gamestates to allow for a main menu