More refactoring. added screens to manage different states of the game. Started input handling