Add some comments to improve the code.