changed the done module -- small change