Update upgrade.rb