added more relationships in models