add relations for some models