Add support for module extension.

Add support for module extension