Add button to delete assignment
Fix issue with weird behavior when not on 1st unit
This got moved to the lesson module.