Added callbacks to store fitted models