added some model relationships