Updating comments in the code