Added functionality to load the data and show an example digit