Update doc comment