Add comment to code