Add support for custom modules