diff --git a/machine-learning-ex2/ex2/predict.m b/machine-learning-ex2/ex2/predict.m
index 34a2905..c393e0a 100644
--- a/machine-learning-ex2/ex2/predict.m
+++ b/machine-learning-ex2/ex2/predict.m
@@ -16,6 +16,10 @@
 %
 
 
+h = sigmoid(X * theta); 
+
+p(find(h > 0.5)) = 1;
+p(find(h < 0.5)) = 0;