Slightly improved the comments