small edits, including a fix for figure(2)