Add a after_build callback to OF