Added code for experiment 2