Python correction