Small tweaks to the explanations of the code