Adding logic to read training folder directly. No need for csv now.