Minor tweaks to schematic diagram