Update comments

Updated comments