Add upgrade script