correcting type signature on classify