Slight modifications to esc key behavior