Add a wait for buttons