updating code
just edits

more edits