Write some simple test for model