Add six for py3/py2 compatibility