Update SOLUTION IN PYTHON