I think it's probably time to remove anything that's Python 2 specific from this example. (cc @hugovk?)