Fixed code, used states to exit game, return to menu on loss