Update jet_metric.py