new loss; figure is generated using Linear.py