Add max_wait param