Improve the auto-updater