Added a hook to post-process models