Added output of individual unit losses