Added state handeling and start - game - gameover screens