python code for company