Add custom functions on button press