Flux and Gauss's law in code | StemPy Lab