Correct & improve the example code in the readme.