Add code to plot the atlas