Small Edits to plotting functions