Added lose condition and tweaked menu