Added tests for Model class