1. Created a Screen class to handle drawing different screens (game
screen, main menu screen, etc)
2. Some code clean-up