Update docs to match with the implementation