I would have probably raised a bit more pre-flop. I always like to be heads up with AA or KK. Just me though.