Cooling of drive motors