added condition to extend the module