Update 3d_density.py