The model is now working