added unit testing for models