took out seaborn and made small changes