Adding an improved plot using seaborn