Modified migration script to not remove columns