Wrote the bulk of compute_grid_of_number_densities code! seems to work.
Cleaned up comments/docstrings a bit.