Add relations to models