Update migration class