Add after_build_models template method