adding unit tests for model