diff --git a/.buildinfo b/.buildinfo
index 6a3bf912..dc5b210e 100644
--- a/.buildinfo
+++ b/.buildinfo
@@ -1,4 +1,4 @@
 # Sphinx build info version 1
 # This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
-config: eae5f821a6a1a15c65bb7d9d13ab6bbf
+config: c9010c591cbe7dcee37916934400929a
 tags: 645f666f9bcd5a90fca523b33c5a78b7
diff --git a/.doctrees/environment.pickle b/.doctrees/environment.pickle
index 3363a3dc..7d6ae951 100644
Binary files a/.doctrees/environment.pickle and b/.doctrees/environment.pickle differ
diff --git a/.doctrees/python-api/generated/pyjuice.structures.PD.doctree b/.doctrees/python-api/generated/pyjuice.structures.PD.doctree
index 998fc07a..0a96cf39 100644
Binary files a/.doctrees/python-api/generated/pyjuice.structures.PD.doctree and b/.doctrees/python-api/generated/pyjuice.structures.PD.doctree differ
diff --git a/_downloads/662999063954282841dc90b8945f85ce/tutorials_jupyter.zip b/_downloads/662999063954282841dc90b8945f85ce/tutorials_jupyter.zip
index 4affb25c..814cd1fe 100644
Binary files a/_downloads/662999063954282841dc90b8945f85ce/tutorials_jupyter.zip and b/_downloads/662999063954282841dc90b8945f85ce/tutorials_jupyter.zip differ
diff --git a/_downloads/763344228ae6bc253ed1a6cf586aa30d/tutorials_python.zip b/_downloads/763344228ae6bc253ed1a6cf586aa30d/tutorials_python.zip
index 563b1b9b..0ad82a62 100644
Binary files a/_downloads/763344228ae6bc253ed1a6cf586aa30d/tutorials_python.zip and b/_downloads/763344228ae6bc253ed1a6cf586aa30d/tutorials_python.zip differ
diff --git a/python-api/generated/pyjuice.structures.PD.html b/python-api/generated/pyjuice.structures.PD.html
index 4a6ff048..92344b9f 100644
--- a/python-api/generated/pyjuice.structures.PD.html
+++ b/python-api/generated/pyjuice.structures.PD.html
@@ -101,7 +101,7 @@
 <h1>pyjuice.structures.PD<a class="headerlink" href="#pyjuice-structures-pd" title="Link to this heading"></a></h1>
 <dl class="py function">
 <dt class="sig sig-object py" id="pyjuice.structures.PD">
-<span class="sig-prename descclassname"><span class="pre">pyjuice.structures.</span></span><span class="sig-name descname"><span class="pre">PD</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">data_shape:</span> <span class="pre">Tuple</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">num_latents:</span> <span class="pre">int</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">split_intervals:</span> <span class="pre">Optional[Union[int</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">Tuple[int]]]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">split_points:</span> <span class="pre">Optional[Sequence[Sequence[int]]]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_split_depth:</span> <span class="pre">Optional[int]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_prod_block_conns:</span> <span class="pre">int</span> <span class="pre">=</span> <span class="pre">4</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">structure_type:</span> <span class="pre">str</span> <span class="pre">=</span> <span class="pre">'sum_dominated'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_layer_fn:</span> <span class="pre">Optional[Callable]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_dists:</span> <span class="pre">Optional[Distribution]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_layer_type:</span> <span class="pre">Type[Distribution]</span> <span class="pre">=</span> <span class="pre">&lt;class</span> <span class="pre">'pyjuice.nodes.distributions.categorical.Categorical'&gt;</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_layer_params:</span> <span class="pre">Dict</span> <span class="pre">=</span> <span class="pre">{'num_cats':</span> <span class="pre">256}</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">use_linear_mixing:</span> <span class="pre">bool</span> <span class="pre">=</span> <span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">block_size:</span> <span class="pre">Optional[int]</span> <span class="pre">=</span> <span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#pyjuice.structures.PD" title="Link to this definition"></a></dt>
+<span class="sig-prename descclassname"><span class="pre">pyjuice.structures.</span></span><span class="sig-name descname"><span class="pre">PD</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">data_shape:</span> <span class="pre">Tuple</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">num_latents:</span> <span class="pre">int</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">split_intervals:</span> <span class="pre">Optional[Union[int</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">Tuple[int]]]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">split_points:</span> <span class="pre">Optional[Sequence[Sequence[int]]]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_split_depth:</span> <span class="pre">Optional[int]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_prod_block_conns:</span> <span class="pre">int</span> <span class="pre">=</span> <span class="pre">4</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">structure_type:</span> <span class="pre">str</span> <span class="pre">=</span> <span class="pre">'sum_dominated'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_layer_fn:</span> <span class="pre">Optional[Callable]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_dist:</span> <span class="pre">Optional[Distribution]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_layer_type:</span> <span class="pre">Type[Distribution]</span> <span class="pre">=</span> <span class="pre">&lt;class</span> <span class="pre">'pyjuice.nodes.distributions.categorical.Categorical'&gt;</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_layer_params:</span> <span class="pre">Dict</span> <span class="pre">=</span> <span class="pre">{'num_cats':</span> <span class="pre">256}</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">use_linear_mixing:</span> <span class="pre">bool</span> <span class="pre">=</span> <span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">block_size:</span> <span class="pre">Optional[int]</span> <span class="pre">=</span> <span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#pyjuice.structures.PD" title="Link to this definition"></a></dt>
 <dd><p>Generate PCs with the PD structure (<a class="reference external" href="https://arxiv.org/pdf/1202.3732.pdf">https://arxiv.org/pdf/1202.3732.pdf</a>).</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>