minor change to Fig 1c and code to generate necessary basis