Fixing logic to disable movement on game end