Add game over screen and detection