Add comments and minor tweaks