Better key handling for game over