Added columns to migration