Code is more pythonic