Add add_company method