Added unit tests to the model. Tests Passed.