tiny error in code

Hi, thanks for your code.

I think I found some error in your code, but it is so tiny.

~~~python
    # Append new parameter updates to lists.
    mu_n_register.append(a_n)
~~~

~~~python
    # Append new parameter updates to lists.
    mu_n_register.append(mu_n)
~~~

Have a nice day :)