Update doc and TODO