cleaning and added plots