updated migrate class