Add wait for key to start