added more tests to the model
[#8]