Removed unused StringIO import