ModelFitting and plots added