Add pre_build and post_build callback