improved future work description + small edits