Update comments of code