More pythonic style