Made some changes after testing the game