Add relationships to ModelBase