I would re-raise more pre-flop. You raised less than 1/2 pot. I'd make it $6 or $7. And definitely follow through with a c-bet on the flop. With this guy only sitting with 40BB I'm willing to gamble with AKs all day, even when the flop misses me.