Could you add a doc string for the model itself too?