additional edits to code