Fix the comments

Make some of the comments a little more suitable