Ready for a huge bump?

So I did the mod today and hooked the switch up to my 04 and it works like a champ according to the DIC.
Now the next part of the transmission is.... Should I put the 01 harness in my 01 trans or run the 04 harness in the trans?