Fixed documentation to match implementation