more pythonic idiom