Add some relationships to those models