Update to DB upgrading system