a little touch up in Models