Writing Idiomatic Python Video Three code