First model tests

 - Add basic model tests