Revise TODO handling