Improve the doc