added unit tests for some of the models