updated the comments properly