Model associations added