Fix some left behind stuff in the code example in the readme