moved getProbability method to classify
small change to classify