Now the model woooorks!