Update experiment 2 code
Merge branch 'develop'