Add missing StringIO import