I love my KZ-TE! I would start saving for a head now though if it wasn't done when they did the swap. You are on the right track.
I killed my head towing around 2 big quads on a trailer up hills, before I knew about egt and the overheating issues.
Drove it for 4 years until it got bad enough that it was smoking me out and overheating.
I really love the engine and truck though.
Mine has : new AMC head, amc head studs, all new OEM gaskets, separate tranny cooler, new fan hub, fj80 fan, new water pump, toyota red coolant, high flow turbo and injectors from New Zealand, fuel cut defender, boost controller, custom 3"ss dump pipe and exhaust. Going to go with a top mount intercooler with a fan for offroading as well as water injection. .
It definitely has more go then it used to.
Got any questions let me know