Add the upgrade step, tested manually, but needs real tests.

Work in progress for now.