Improve doc

same as above