Use idiomatic Python