 		CUDA_CHECK_RETURN(cudaDeviceSynchronize());
 
 		// set initial values for the characteristics. this will launch a CUDA kernel.
-		InitialValue<ChiType><<<nBlocks,nThreads>>>(p_chi, p_meas, p_meas_unc, p_cholfact, p_logdens_meas);
+		InitialValue<ChiType><<<nBlocks,nThreads>>>(p_chi, p_meas, p_meas_unc, p_cholfact, p_logdens);
 
 		// copy values from device to host
 		h_chi = d_chi;
 		h_cholfact = d_cholfact;
-		h_logdens_meas = d_logdens_meas;
-		h_logdens_pop = d_logdens_pop;
+		h_logdens = d_logdens;
 	}
 
 	virtual ~DataAugmentation() {

	}

	// make sure that the data augmentation knows about the population parameters
	void SetPopulation(PopulationPar& t) {
		Theta = t;
	}


        CUDA_CHECK_RETURN(cudaDeviceSynchronize());
        // transfer values back to host
        h_chi = d_chi;
        h_logdens = d_logdens;
        h_naccept = d_naccept;

        current_iter++;
