New upgrade script