more edits and figures