Add more comments to existing code