Add WaitTime class