Added some basic model tests