Make sure to add model relationships.