Added models class