Add post-gen hook