Add plot image, slight edits