Modify upgrade script to conform to zenrun format