Improved the update module.