Update upgrade script