Updating the UX of the module