When clicking the assignment link or inside the student preview have an edit button available instead of having to back track out of the screen. Having to revert back to the previous page causes the teacher to lose their train of thought and avoids having to reset. I have screencaps to illustrate as well.