     int diag_index = 0;
     for (int j=0; j<dim_theta; j++) {
     	// Initialize the covariance matrix of the theta proposal distribution to be 0.01 * identity(p)
-        theta_cholfactor[j] = 1e-2;
+        theta_cholfactor[diag_index] = 1e-2;
         diag_index += j + 2;
     }
 

    }

    /*
    Set up parallel random number generations on the GPU
    */
    curandState* devStates;  // Create state object for random number generators on the GPU
    // Allocate memory on GPU for RNG states

    //std::ofstream chifile("chis.dat");
    std::ofstream thetafile("thetas.dat");

    int mcmc_iter = 5000;
    int naccept_theta = 0;
    std::cout << "Running MCMC Sampler...." << std::endl;

