add *some* logic for classification