Added some relations to the models