Added relations to models