Andy modified Charlotte's code to make a smaller dataset