use new class for python 2