update the README and the simulation_latent_heterogeneous.py