diff --git a/src/rbm/layer.py b/src/rbm/layer.py
index bc188f5..9afee03 100644
--- a/src/rbm/layer.py
+++ b/src/rbm/layer.py
@@ -120,10 +120,12 @@ def xor():
 	status = Status()
 	layer = RbmLayer("Layer_0", 3, 16, params)
 	layer.init(0.01)
+	layer.load()
 
 	# Train
 	training_batch = np.array([[0,1,1], [0,0,0], [1,1,0], [1,0,1]], dtype=np.float64)
 	layer.train(training_batch, cd_jens, status)
+	layer.save()
 
 	# Test
 	test_batch = np.array([[0,0,0], [0,1,0], [1,0,0], [1,1,0]], dtype=np.float64)