Fixed broken logic of the game (probably previously broken by me)