Add comment to explain some added code