Add code comments to controller