Update format_csv.py

Added `import StringIO module.`