Set correct dtypes for columns