Fixed documentation to line up with code