Fix the plots for linearly separable data