added some citations, fixed future work section