Change in control unit

Changes in control unit according to new design.