Add exit on esc func