1 files changed, 37 insertions, 13 deletions

diff --git a/doc/diagrams/analysisClasses.puml b/doc/diagrams/analysisClasses.puml
index d051cf9..8273930 100644
--- a/doc/diagrams/analysisClasses.puml
+++ b/doc/diagrams/analysisClasses.puml
@@ -1,32 +1,56 @@
 @startuml
 
-!include skinparams.puml
+'!include skinparams.puml
 
-package GameModule {
+() Player
+package "Game module" {
     class GameIO
     class GameState
     class GameBoard
     class GameMove
 
-    GameIO -> GameState
-    GameState -> GameMove
-    GameMove -> GameBoard
+    Player -> GameIO
+    GameIO ..> GameState
+    GameState ..> GameMove
+    GameMove ..> GameBoard
 }
 
-package EngineModule {
+() "Engine user" as EU
+() "Model files" as MF
+package "Engine module" {
     class EngineIO
     class EngineLogic
-    !include DecisionAlgorithm.pumlc
+    interface DecisionAlgorithm
     class MonteCarloTreeSearch
-    class OtherDecisionAlgorithm
+    class MCTSNode
+    class Keras
+    class NeuralNetwork
 
-    EngineIO --> EngineLogic
-    EngineLogic -> DecisionAlgorithm
+    EU -> EngineIO
+    EngineIO ..> EngineLogic
+    EngineLogic ..> DecisionAlgorithm
     DecisionAlgorithm <|.. MonteCarloTreeSearch
-    DecisionAlgorithm <|.. OtherDecisionAlgorithm
+    DecisionAlgorithm <|.. Keras
+    MonteCarloTreeSearch ..> MCTSNode
+    Keras ..> NeuralNetwork
+    NeuralNetwork --> MF
 }
 
-MonteCarloTreeSearch --> GameMove
-OtherDecisionAlgorithm --> GameMove
+() "SGF files" as SGF
+package "Training module" {
+    class Trainer
+    class Parser
+    class ASTNode
+
+    Parser -> SGF
+    Trainer ..> Parser
+    Parser ..> ASTNode
+}
+
+DecisionAlgorithm .> GameMove
+
+ASTNode .> GameMove
+
+Trainer .> NeuralNetwork
 
 @enduml