It would be really great to have a more specific commenting feature rather than just a "reply" feature for assignments.
In Google Docs we can highlight specific passages when commenting, which is incredibly useful.
Making the document editable would also be useful so that students can make changes based on comments
and the dialogue between teacher, TA, and student can progress clearly.