Fix up example for python3