Update split_by_col.py