In general, I like a 2/3 to 1xPSB on the flop. This is better than an open AI push IMO. In this specific example, I don't think it mattered which line was chosen given that that villian didn't have that much more behind.
Folks, let's not forget that AA, and KK are in villian's range (which is of course even wider than that), but we don't KNOW that's what particular hand they are holding. Think in terms of ranges, and not indivdual hands.