added model relationships