Added a ton of new funcitonality to the migration script