Enhance custom module