Fixed PEP8 issue with spacing between function arguments.
Added test for giving `color=None` to plot function.