Added models relations, factories and tests