Updated custom module file