Now handles mark as done and mark as not done, more or less.