Added pressing escape to restart the game