Change TODO python snippet