KDX don't swim

A friend of mine was riding his 2000 KDX 220, slide down a bank and accidentally went into a pond. The water level reached about half way up the gas tank. With it running, he pulled it out of the pond and drove it for about 15 minutes before it shut off on its own. When cold it idles and revs a little. When warmed up it doesn't rev or idle at all. So far they have pulled and cleaned the carburetor, and replaced the fuel. He also switched his KDX 220 Carburetor (really a 200) with his 2003 KDX 200 carburetors. It still runs the same. When he pulled off the side cover, water poured out from where the stator is. He dried out the area with a hair drier. He doesn't know if the stator or the electronic module under the seat (also submerged under water) may be the problem. Being that both of these items are costly, he didn't know which one he should replace first. Also, is it safe and logical to try the KDX 200 stator and electronic module on his 220?
Thanks so much for your help.
Ps, Remember, KDX don't swim.