Added better input handling and game screens