Add max_wait parameter