Add module loading support