Update format_csv.py

Work in progress...