add classify function