change code for adding comments