Added some paragraphs about future work, a results figure, and updated affiliation