Added upgrade steps