[enh] Create upgrade script