Done! *insert thumbs-up emoji*
Nice you were able to add a restart, but if you want to make a lose state you could add a timer