Add python 2 compatibility