Edits to future work & discussion