Added linewidth argument to plotting functions