Update old Python code