Fixing issues
Fix train classifier