Should probably remove the StringIO import.