adding test classify code