Make example ready for python3