Update criterion.py