Added test for done function, first implementation ready