If memory serves, it was never explained how Kay returned to the agency, but it was never really an issue as Kay and Jay remained partners, with Elle working in the lab.
I wish you could have seen the other seasons, as the show went through a little bit of changes from season 1 to 2, and then...