Added a color argument for publication-ready plots