Add new tests for model