Adding upgrade migration