Yes, an immediate restart would be a great addition.
Thank you for the kind words, I always make my games simple and easy to understand and play, especially during game jams, and I think it does add to the fun when you play the game smoothly without any interruptions.