- [ ] Add tests (heat loss test) - [ ] Add more diffusion functions with test (test should be the output being between 0 and 1) - [ ] Add more laplacian stencils (related to another PR) - [ ] Better docs - [ ] Better demo