Added migrate method