def METHOD_NAME(self, total_density):
    """Sets density.
    Sets the density in the exact same order as total_density_list outputs,
    allowing for internal consistency
    Parameters
    ----------
    total_density : list of numpy.ndarray
        Total atoms.
    """
    for i, density_slice in enumerate(total_density):
        self.set_mat_slice(i, density_slice)