Fixed a bit of comments