Updated the module wiht better ux