improve doc comments