Modify codes for experiment