Optimizing Python Code - DV