now adding correct class training list instead of feature training list again