fixing classifier method