Added tests for model