add tests for models