Fixed the code example in the readme.