Fixed NBC to take training folders as input.
Small fix.