Update losses.py