Updates to the OM upgrade script.