Fix document in code

I fixed some mistakes in code comment.