Please note there is a bug - when you lost you get stuck if you don't return to the main menu (ESC)
I found the bug and was able to fix it, so now it should be fine. please let me know if you find other bugs.