added relations in models