adding unit test for model building