The Droid control ship has a unique design. From an attack perspective it’s the equivalent of the thermal exhaust port on the Death Star. I would have kept the other capital ships visible in the battle. The Federation would be so convinced of its victory that the fighters launched from the control ship would have been enough. Visually you don’t have the threat suddenly reduced. Once the control ship is down they can’t launch anymore fighters as they were controlled from the control ship. You might have them try to escape like in the Battle of Scarif only to have Republic ships turn up and stop some (once Palpatine has seen the tide turn). The battle doesn’t currently feel right. Which a problem with the original film and not your edit.