1 . Add init upgrade code