added code to make simulation figures in paper