Finish up upgrade script