Added model association