Kai's presence in Season 2

Contains minor spoilers:

I was really disappointed to see that Kai, who I found to be one of, if not the most badass character in the show, had zero presence after Kei runs and steals his phone in episode 2 or 3. Will he return or has he been completely forgotten about?