Updated docs to match code