update code and figures for JSM talk