amend main script - create figure