fix bug and add migration
add migration