Minor changes for the experiment