minor corrections in plot