improving classification model/methods. wip