Added code that loads the data and generates the visualizations