A little more Pythonic.