diff --git a/source/main.cpp b/source/main.cpp
index a2e9be9..8779d85 100644
--- a/source/main.cpp
+++ b/source/main.cpp
@@ -90,22 +90,15 @@ int main()
 	printf("Loaded %d training samples\n", (int)numTraining);
 
 #if 0
-	size_t numVisibleX = 16;
-	size_t numVisibleY = 16;
-	size_t numHidden = 256;
-	
-	int i=0;
-	Layer *lowerLayer = new Layer("Layer", i, numVisibleX, numVisibleY, numHidden);
+	int i = 0;
+	Layer *lowerLayer = new Layer("Layer", i, 16, 16, 256);
 	stack.addLayer(lowerLayer);
-	numHidden >>= 1;
 	
-	i++;
-	for (i; i < 1; i++)
+	for (++i; i < 1; i++)
 	{
-		Layer *layer = new Layer("Layer", i, lowerLayer->bh().n_elem, 1, numHidden);
+		Layer *layer = new Layer("Layer", i, lowerLayer->bh().n_elem, 1, lowerLayer->bh().n_elem >> 1);
 		lowerLayer = layer;
 		stack.addLayer(layer);
-		numHidden >>= 1;
 	}
 
 	// Save project