From 834c2f348422a46f542263127de27d8a915d00a5 Mon Sep 17 00:00:00 2001
From: Antonio Andriella <aandriella@iri.upc.edu>
Date: Wed, 4 Aug 2021 17:12:48 +0200
Subject: [PATCH] update documentation

---
 README.md | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index e65c6ae..8dc6c91 100644
--- a/README.md
+++ b/README.md
@@ -1 +1,9 @@
-# task_environment
+### Auxiliary package to formalise the cognitive training domain ### 
+
+#### Package:
+- the folder contains the following classes:
+    * cognitive_game_env.py defines the cognitive domain used in the IRL algorithm
+    * cognitive_game_vars.py defines the variables employed by the BNs
+    * environment.py defines the main rules of the cognitive exercise
+    * episode.py, class to initialise and store the episodes (s, a, s')
+    
\ No newline at end of file
-- 
GitLab