* Added unit testing for models