More functionality in migration script.