Much better comments and function organization, improved UI for game over