Could it be the replacement loom is different at the h/light switch terminals, and not working with new switch? I know there's a few different types that interconnect, but can short things out and cause fire/ problems. I'm thinking this as the wiring is all plug + play, and fairly simple to do without going wrong. Ie plug things in exactly where they were, with labelling etc if you really can't remember or work it out. Surely if headlight function was fine before all this, then I'd check terminals marry up + supply the same cables on both looms on the red h/light socket