Solving Problems with Python Walrus Operator