improve doc comment