Fix plot
Fix experiment