Added upgrade script