Fixed docs to match code