Added functionality to my experiment