Add support for custom action on menu click