Added unit tests for company model.