Fixed python 3 support