TST: Fixed unittests.
DOC: Reran demo simulations. Updated demo.rst.