- Optimize the way to retrieve class with maximum confidence. Avoir multiple pass. - Add support for label files