Corrected the model