Add: new types to Upgrade script