diff --git a/ex2/plotData.m b/ex2/plotData.m
index 5df9d3f..4e24c87 100644
--- a/ex2/plotData.m
+++ b/ex2/plotData.m
@@ -1,5 +1,5 @@
 function plotData(X, y)
-%PLOTDATA Plots the data points X and y into a new figure 
+%PLOTDATA Plots the data points X and y into a new figure
 %   PLOTDATA(x,y) plots the data points with + for the positive examples
 %   and o for the negative examples. X is assumed to be a Mx2 matrix.
 
@@ -11,7 +11,11 @@ figure; hold on;
 %               2D plot, using the option 'k+' for the positive
 %               examples and 'ko' for the negative examples.
 %
+pos = find(y==1);
+neg = find(y==0);
 
+plot(X(pos, 1), X(pos, 2), 'k+');
+plot(X(neg, 1), X(neg, 2), 'ko');