Try to support python3

Python2 of course should still work.