Update figure_2.R

Make compatible with clean formatting of Dryad submission.