Added done method