Updated plot_lines to accept a color argument.