Transmission cooler for 8sp and K04?

I run my car pretty hard at the track and started thinking about how much extra abuse the K04 can dish out to the tranny. I talked to a local shop they suggested I install a transmission cooler, they said the stock unit is integrated with the main cooling system and not independent. I could have misunderstood them but just the same, should one consider adding a cooler?

Secondary question: what about oil coolers. IS the stock one big enough should it be upgraded? I was hitting around 244f last Friday at the track according to my laptimer\oiltemp VCDS mod. The air temp was around 80f.