A bit more pythonic