Fixed classifier. Returns correct classes now