Add support for custom actions