Add wait until butten press