Update custom_metric.go