Updated circuit diagram for main unit