Adding training metrics