Added models and relation