Refactor modifications in custom module.