Added models tests