added classification fcn which somehow works pretty well