how to write pythonic code