Add support for module statement