Two little model changes