Added tests for models