Press space button to stop and then press the flag again to restart.
This was just a quick thing so it isn't the best code in the world. Please feel free to put a comment down below.
