diff --git a/Source/LayerArray.hpp b/Source/LayerArray.hpp
index 4fb4832..45218f0 100644
--- a/Source/LayerArray.hpp
+++ b/Source/LayerArray.hpp
@@ -205,6 +205,9 @@ public:
 		{
 			uint32_t numUnits;
 			fscanf(pFile, "%d\n", &numUnits);
+			if (numUnits == 0)
+				break;
+
 			pData = new double[numUnits];
 			T* data = add(nullptr, numUnits);
 
diff --git a/Source/MainComponent.cpp b/Source/MainComponent.cpp
index 76cc9bc..a64fd08 100644
--- a/Source/MainComponent.cpp
+++ b/Source/MainComponent.cpp
@@ -482,7 +482,7 @@ void MainComponent::buttonClicked (Button* buttonThatWasClicked)
     else if (buttonThatWasClicked == ShakeButton)
     {
         //[UserButtonCode_ShakeButton] -- add your button handler code here..
-    	m_weights.shuffle(0.01);
+    	m_weights.shuffle(0.001);
     	redrawReconstruction();
     	redrawWeights((int)WeightsSlider->getValue());
         //[/UserButtonCode_ShakeButton]