wrap up if statement, use setTime function