Add clarifying comments to the code