Removed that break that trained only few files, now it trains everything