Changed plotting in experiment