Game working but problems going back after a square