Add new APIs to set/get the upgrade status