Using wait_for_keyboard instead of await_keyboard, for consistency