Added loss plot