Structure of code improved. Separate the game into different screens.