Improved the comments