Added relational stuff to models