support python 3 as well