Add 'import io' in util.py