Make it more pythonic.