Trying to integrate the button