Made some improvements to the update system.