pythonic way to write code