I would 3-bet preflop to define my hand aswell as his. I have a hard time seeing him 3-bet a flop with those cards if I 3-bet preflop unless he actually have a hand. Now, I am not sure if he is trying to steal it once again but his 3-bet is worrying me.

I like the line after the flop since it got him scared.

Only thing I would make different is to 3-bet preflop.