Extract out classifier. Refactor probability.