Added code for running experiments