Add custom functionality to disable function