Update company.py