Fix example in readme to work with py3