made extra edits to code