removed the import for six package. not sure if this is needed