Improve example in readme

Hi @dhagrow, can you please submit a PR for this?
Thanks