Be more pythonic