Work in progress
Fixed bug in _initialize_sol
Started on a demo notebook for simple_diffusion