Our "game over" message says you can press 'S' to restart, but we haven't done that bit yet. So here it is:
- Define a function I've chosen to name restart_game which sets all our key variables up to how they should be at the start. In particular, notice that we set the bats array back to empty, and we call AddNewBat to get the bats started again.
- Alter the keydown handler to check for 'S' and, if the game was over, call restart_game
- Finally, instead of calling AddNewBat once, unconditionally to get things started, we instead call reset_game. Much neater.