Adding a wait control