fix code after review
update doc