Make the example run also on Python 3