Added more control options to the upgrade script