diff --git a/src/session/controller_itp.ml b/src/session/controller_itp.ml
index 5b935477ba8a2f6e54dea380117d37ef5c0c8e70..63657c2115fb174338114f5f0d82b93932ee7cd4 100644
--- a/src/session/controller_itp.ml
+++ b/src/session/controller_itp.ml
@@ -25,26 +25,90 @@ let print_status fmt st =
   | InternalFailure e -> fprintf fmt "InternalFailure(%a)" Exn_printer.exn_printer e
 
 type transformation_status =
-  | TSscheduled | TSdone | TSfailed
+  | TSscheduled | TSdone of transID | TSfailed
 
 let print_trans_status fmt st =
   match st with
   | TSscheduled -> fprintf fmt "TScheduled"
-  | TSdone -> fprintf fmt "TSdone"
+  | TSdone tid -> fprintf fmt "TSdone" (* TODO print tid *)
   | TSfailed -> fprintf fmt "TSfailed"
 
+open Ident
+
+type proof_state = {
+    th_state: bool Hid.t;
+    tn_state: bool Htn.t;
+    pn_state : bool Hpn.t;
+  }
+
+let init_proof_state () =
+  {th_state = Hid.create 7;
+   tn_state = Htn.create 42;
+   pn_state = Hpn.create 42}
+
 type controller =
   { mutable controller_session : Session_itp.session;
+    proof_state : proof_state;
     controller_env : Env.env;
     controller_provers : (Whyconf.config_prover * Driver.driver) Whyconf.Hprover.t;
   }
 
 let create_controller env s = {
     controller_session = s;
+    proof_state = init_proof_state ();
     controller_env = env;
     controller_provers = Whyconf.Hprover.create 7;
 }
 
+let find_tn c tid =
+  try Htn.find c.proof_state.tn_state tid with
+  | Not_found -> false
+
+let find_pn c id =
+  try Hpn.find c.proof_state.pn_state id with
+  | Not_found -> false
+
+(* Update the result of the theory according to its children *)
+let update_theory th ps =
+  let goals = theory_goals th in
+  Hid.replace ps.th_state (theory_name th)
+    (List.for_all (fun id -> Hpn.find ps.pn_state id) goals)
+
+let rec propagate_proof c (id: Session_itp.proofNodeID) =
+  let tr_list = get_transformations c.controller_session id in
+  let new_state = List.exists (fun id -> find_tn c id) tr_list in
+  if new_state != find_pn c id then
+    begin
+      Hpn.replace c.proof_state.pn_state id new_state;
+      update_proof c id
+    end
+
+and propagate_trans c (tid: Session_itp.transID) =
+  let proof_list = get_sub_tasks c.controller_session tid in
+  let cur_state = find_tn c tid in
+  let new_state = List.for_all (fun id -> find_pn c id) proof_list in
+  if cur_state != new_state then
+    begin
+      Htn.replace c.proof_state.tn_state tid new_state;
+      propagate_proof c (get_trans_parent c.controller_session tid)
+    end
+
+and update_proof c id =
+  match get_proof_parent c.controller_session id with
+  | Theory th -> update_theory th c.proof_state
+  | Trans tid -> propagate_trans c tid
+
+(* [update_proof_node c id b] Update the whole proof_state
+   of c according to the result (id, b) *)
+let update_proof_node c id b =
+  Hpn.replace c.proof_state.pn_state id b;
+  update_proof c id
+
+(* [update_trans_node c id b] Update the proof_state of c to take the result of (id,b). Then
+   propagates it to its parents *)
+let update_trans_node c id b =
+  Htn.replace c.proof_state.tn_state id b;
+  propagate_proof c (get_trans_parent c.controller_session id)
 
 
 module type Scheduler = sig
@@ -136,19 +200,28 @@ let run_timeout_handler () =
       S.timeout ~ms:125 timeout_handler;
     end
 
-let schedule_proof_attempt c id pr ~limit ~callback =
+let schedule_proof_attempt_r c id pr ~limit ~callback =
   graft_proof_attempt c.controller_session id pr ~timelimit:5;
   Queue.add (c,id,pr,limit,callback) scheduled_proof_attempts;
   callback Scheduled;
   run_timeout_handler ()
 
-let schedule_transformation c id name args ~callback =
+let schedule_proof_attempt c id pr ~limit ~callback =
+  let callback s = (match s with
+  | Done pr -> update_proof_node c id true
+  | Interrupted | InternalFailure _ -> update_proof_node c id false
+  | _ -> ());
+  callback s
+  in
+  schedule_proof_attempt_r c id pr ~limit:limit ~callback:callback
+
+let schedule_transformation_r c id name args ~callback =
   let apply_trans () =
     let task = get_task c.controller_session id in
     try
       let subtasks = Trans.apply_transform_args name c.controller_env args task in
-      let _tid = graft_transf c.controller_session id name args subtasks in
-      callback TSdone;
+      let tid = graft_transf c.controller_session id name args subtasks in
+      callback (TSdone tid);
       false
     with e when not (Debug.test_flag Debug.stack_trace) ->
       Format.eprintf
@@ -160,6 +233,13 @@ let schedule_transformation c id name args ~callback =
   S.idle ~prio:0 apply_trans;
   callback TSscheduled
 
+let schedule_transformation c id name args ~callback =
+  let callback s = (match s with
+  | TSdone tid -> update_trans_node c tid false (*(get_sub_tasks c.controller_session tid = [])*) (* TODO need to change schedule transformation to get the id ? *)
+  | TSfailed -> ()
+  | _ -> ()); callback s in
+  schedule_transformation_r c id name args ~callback:callback
+
 let read_file env ?format fn =
   let theories = Env.read_file Env.base_language env ?format fn in
   let ltheories =
diff --git a/src/session/controller_itp.mli b/src/session/controller_itp.mli
index b8edc44f00b9e737a1e4caadc547a694a8d8fe14..beb52c9bb8eebf8d3b186e3bcf24f1c1a6a92c97 100644
--- a/src/session/controller_itp.mli
+++ b/src/session/controller_itp.mli
@@ -23,7 +23,7 @@ type proof_attempt_status =
 
 val print_status : Format.formatter -> proof_attempt_status -> unit
 
-type transformation_status = TSscheduled | TSdone  | TSfailed
+type transformation_status = TSscheduled | TSdone of transID  | TSfailed
 
 val print_trans_status : Format.formatter -> transformation_status -> unit
 
@@ -51,14 +51,36 @@ module type Scheduler = sig
 
 end
 
+open Ident
+
+(** Correspondance between a node of the proof tree
+    and its state (proved or not) *)
+type proof_state = {
+    th_state: bool Hid.t;
+    tn_state: bool Htn.t;
+    pn_state : bool Hpn.t;
+  }
+
 type controller = private
   { mutable controller_session : Session_itp.session;
+    proof_state : proof_state;
     controller_env : Env.env;
     controller_provers : (Whyconf.config_prover * Driver.driver) Whyconf.Hprover.t;
   }
 
 val create_controller : Env.env -> Session_itp.session -> controller
 
+(** [update_proof_node c id b] Update the whole proof_state
+   of c according to the result (id, b) *)
+val update_proof_node: controller -> Session_itp.proofNodeID -> bool -> unit
+
+(** [update_trans_node c id b] Update the whole proof_state of c
+   according to the result (id,b) *)
+val update_trans_node: controller -> Session_itp.transID -> bool -> unit
+
+(** Used to find if a proof node is proved or not TODO remove *)
+val find_pn: controller -> Session_itp.proofNodeID -> bool
+
 module Make(S : Scheduler) : sig
 
 val schedule_proof_attempt :
diff --git a/src/session/session_itp.ml b/src/session/session_itp.ml
index a7f1fdccd6e034bc974e2793cc951ea756046136..3c682038ba2606571e54909aebee473873e3598c 100644
--- a/src/session/session_itp.ml
+++ b/src/session/session_itp.ml
@@ -57,6 +57,23 @@ type file = {
   file_theories : theory list;
 }
 
+module Proofnodeid = struct
+  type t = proofNodeID
+  let compare (x : proofNodeID) y  = Pervasives.compare x y
+  let equal (x : proofNodeID) y = x = y
+  let hash  (x : proofNodeID) = x
+end
+
+module Transid = struct
+  type t = transID
+  let compare (x : transID) y  = Pervasives.compare x y
+  let equal (x : transID) y = x = y
+  let hash  (x : transID) = x
+end
+
+module Hpn = Exthtbl.Make(Proofnodeid)
+module Htn = Exthtbl.Make(Transid)
+
 type session = {
   proofNode_table               : proof_node Hint.t;
   mutable next_proofNodeID      : int;
@@ -67,6 +84,17 @@ type session = {
   session_prover_ids            : int Hprover.t;
 }
 
+(*
+(* TODO replace *)
+let init_Hpn (s : session) (h: 'a Hpn.t) (d: 'a) : unit =
+  Hint.iter (fun k _pn -> Hpn.replace h k d) s.proofNode_table
+
+let init_Htn (s : session) (h: 'a Htn.t) (d: 'a) : unit =
+  Hint.iter (fun k _pn -> Htn.replace h k d) s.trans_table
+*)
+
+
+
 let session_iter_proofNode f s =
   Hint.iter
     (fun id pn -> if id < s.next_proofNodeID then f pn)
diff --git a/src/session/session_itp.mli b/src/session/session_itp.mli
index 25b698cd5a4fffd9d8c7fecfa1c7e676bacc621d..55c12de186eadbd8607b276e362df3075f464b3d 100644
--- a/src/session/session_itp.mli
+++ b/src/session/session_itp.mli
@@ -17,6 +17,11 @@ unique identifiers of type [proofNodeId]
 type session
 
 type proofNodeID
+type transID
+
+module Hpn: Exthtbl.S with type key = proofNodeID
+module Htn: Exthtbl.S with type key = transID
+
 
 type theory
 
@@ -49,7 +54,6 @@ type proof_attempt = {
   proof_obsolete      : bool;
   proof_script        : string option;  (* non empty for external ITP *)
 }
-type transID
 
 
 type proof_parent = Trans of transID | Theory of theory
diff --git a/src/why3shell/why3shell.ml b/src/why3shell/why3shell.ml
index 9b48c0ae652dc5b68a6cc80c366a6d0f187f9b3f..e9517ab096fa3977b899f4d080be01a934af9a67 100644
--- a/src/why3shell/why3shell.ml
+++ b/src/why3shell/why3shell.ml
@@ -365,8 +365,16 @@ let rec print_proof_node s (fmt: Format.formatter) p =
     | Some pn -> pn = p
     | _ -> false
   in
+  (* TODO proved or not to be done on the whole tree. Just a test *)
+  let is_proved () =
+    match is_goal_cursor () with
+    | Some p -> Controller_itp.find_pn cont p
+    | _ -> false
+  in
   if current_goal then
     fprintf fmt "**";
+  if is_proved () then
+    fprintf fmt "P";
 
   fprintf fmt
     "@[<hv 2> Goal %s; parent %s;@ @[<hov 2>[%a]@]@ @[<hov 2>[%a]@]@]"
@@ -483,7 +491,7 @@ let test_transform_and_display fmt args =
          fprintf fmt "transformation status: %a@."
                  Controller_itp.print_trans_status status;
          match status with
-         | TSdone -> (ignore (move_to_goal_ret next_node);
+         | TSdone tid -> (ignore (move_to_goal_ret next_node);
              test_print_goal fmt [])
          | _ -> ()
        in