changes from re-reunning to confirm results
made notebook to keep track of what's in the dataset and what's not