correcting logic

js/script.js

@@ -14,6 +14,7 @@ grid(6, 6, '.container');
 
 
 var arr = [];
+var nwarr = [];
 
 $('td').on('click', function(event) {
 
@@ -30,10 +31,18 @@ $('td').on('click', function(event) {
 });
 
 $('#clear').on('click', function(){
-	$('td').css({'background' : '', 'height' : '30px', 'width' : '30px'});
+	for (i=0, j=0; i< arr.length; i++) {
+		if ($(arr[i]).hasClass('black') ) {
+			nwarr[j] = arr[i];
+			j++;
+		}
+	}
+	$(nwarr).removeClass('black').addClass('white');
 });
 
 
 $('#restore').on('click', function(){
-	$(arr).css({'background' : '#000', 'height' : '30px', 'width' : '30px'});
-});
+
+	$(nwarr).removeClass('white').addClass('black');
+	nwarr =[];
+});