Trying to make things look a little more pythonic.