updated the module to work properly