Upgrade of assembly equipment