Clean up README code example

Hey man, don't know if this is the latest version of the repo, but I figured I would pull request the changes I had to make to get the code example in the README running.