Added migration for company_id