Added more columns to nan_values in model.py