updating plotting code and README