I like what you say about the lateral movement. If I were Justin I would not follow Khabib around the cage. Either blitz him with a quick combo that ends with a power strike or a takedown such as 1,2,1,shoot or 1,2, 3, then circle out or make Khabib follow him so he can plant and attack when the opportunity is there.
I kinda like Justin's mindset as in bring the pain, go to war etc. But the mindset of defensive wrestling is going to hurt him imo. I would rather see a mindset of I am going to attack. Then he needs to go in there and attack, at first with offensive wrestling imo and then let that open up the striking opportunities. Khabib stays offensive because he carries this I am going to attack mindset but he fights smart. He needs to fight like Khabib imo.
I won't comment on what I would like to see once he gets to the ground because I have no idea what he is capable of. Khabib on top is a nightmare and his bottom game is great too. Can Justin spend time in his guard and do a little damage? I don't know the answer to that. Can Justin stand back up if he is on top? He should be able to. If he stays down though Khabib does have submissions and setups though so be careful.
The main reason why I oppose any offensive wrestling from Justin beyond low commitment knee taps is 1) I don't think Khabib is afraid of being on his back against Justin, so I don't think it'll mess up Khabib's rhythm too bad and 2) I don't think being engaged in full fledged grappling is good for Justin.