Finishing off the upgrade system.