Update mondrian.py