Classify v2

need to make sure it works