Added some testing for company model