Adding another delegate method for returning the button