Adding some more migrate stuff