[DOC] Updated doc comments