Add new tests in the company.feature and modification of the old ones to fit the new address entity