added a dialog button as well