Fills data_density function; updates tests