Fix comments in documentation