Added code for custom button