Import six

Unless `import six` is added, I get this error `NameError: global name 'six' is not defined`