Python code example as requested by a company