Add confirmation on button as well