Add use_migration flag and refactor shell-migrate