more edits
added more code