add wait for keypress