Remade to StateBasedGame, added a main menu
Small codefixes