Adding comment to code