adds unit testing for models