Add ability to mark as done