Yellowstone tv show

I'm catching up on some other shows I watch and saw where the guy that plays Jimmy (Jefferson White) on Yellowstone is on Chicago PD this season.
He's playing a recovering addict who is the police chief's son and is running a treatment home for teenage homeless kids and is suspected of being
the head of a child sex ring. Got me wondering if he comes back to Yellowstone.
He will be starring in the spin-off based on the ranch he got shipped off to and left for, the Four Six ranch 6666.