The model is finally working as intended