Test mark_as_done() function