Fix comments

Fixed some comments