Add migration script