Update comments in code