Updated notebook with better figure labelling