Fix arg name in example

[jdcc]

Just simple doc fixes for those of us who follow along at home. I think I linked to the paper you were referencing, but correct me if I'm wrong.

[pablobarbera]

Thank you!!