Change key presses to work better between levels