Modify the code for the doc