Small fixes
Plot update