add a game over screen and a restart button
Just updated, now it features a main menu and game over screen. Thanks for the feedback!