add tests for simple model