Add wait param to start_driver