Added some models and relations