Add wait.go