Abstracted a little bit so that the classification task can be set to a different tag type more easily.