Updated docs to match code changes