Refactored code to fetch the class name of the module.