added some basic classification stuff