Add migrate as an optional parameter