Add some comments to existing code