Remove unneeded example code
Update migration handler