adding models tests in class