I think he will come good over time working with Klopp. A lot has changed for him in terms of the style of play, position, talent surrounding him and now even the manager. If he is willing to improve then Klopp will improve him. If what happened at the touchline happens at Melwood too and Milner still goes about playing in a way that Klopp does not wish for us to do, he will be shipped out.