updated experiment, made separate class for experiment setup