Add waiting on response after button push