Fix the issue with plot image resizing
Try putting experiment init in try/catch
Set ylim to [0, 1]