Update comments if necessary