Add some models tests. alown was right, it's ugh