Welcome to the forum Poppy.
Stalling could be caused from any number of issues, or possibly one single unit.
Suggestions from members is, install a clear plastic hose 1/4”, at the front of the injection pump to the fuel return pipe. Watch for air bubbles blowing through the clear hose.
Then there is the ever famous PMD, pump mounted driver, original factory location is on the side of the injection pump, unless it has been relocated. Look on the left side of the pump, if there is a box hanging on there, and, if the wire harness is still connected to it, then a new PMD and harness should be installed, a harness long enough to get it relocated to behind the front bumper.
Check with Leroy Diesel for the complete kit with the proper heat sink.
Leroy is a vendor that helps to keep this site supported.
Another vendor is Quad Star Tuning and I believe He too has the Flight Systems kits.