updated documentation to fit example code