Use wait argument from user