added comment for clarity
annotated
fixed bug in migration. added more tests