switched from seaborn to matplotlib