Collimation of energy in medium-modified QCD jets