Add basic model tests