Updated docstring
Corrected error in example code