Added in gamestates and a menu