Update README.md so example works out of the box

I was trying this out and it errored because __init__ didn't have enough arguments.