Added a game over screen, changed the main function