Added callback method for handling button click