Fix comments a little bit