diff --git a/src/main.py b/src/main.py
index a29b0c8..01a0f04 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,29 +1,31 @@
 from service.dialog_state_monitor import DialogStateMonitor
 from service.dialog_policy import DialogPolicy
 from service.natural_languag_understanding import NaturalLanguageUnderstanding
-from service.natural_language_generation import NaturalLanguageGeneration
+from service.natural_language_generation import NaturalLanguageGeneration, parse_frame
+from service.templates import templates
 
 # initialize classes
 nlu = NaturalLanguageUnderstanding() # NLU
 monitor = DialogStateMonitor() # DSM
 dialog_policy = DialogPolicy() # DP
-language_generation = NaturalLanguageGeneration() # NLG
+language_generation = NaturalLanguageGeneration(templates) # NLG
 
 # Main loop
 user_input = input("Możesz zacząć pisać.\n")
 while True:
     # NLU
     frame = nlu.process_input(user_input)
-    print(frame)
+    # print(frame)
 
     # DSM
-    monitor.append(frame)
+    # monitor.append(frame)
 
     # DP
-    print(dialog_policy.next_dialogue_act(monitor.get_all()).act)
+    # print(dialog_policy.next_dialogue_act(monitor.get_all()).act)
 
     # NLG
-    response = language_generation.respond_to_name_query("Jak masz na imię?")
+    act, slots = parse_frame(frame)
+    response = language_generation.generate(act, slots)
     print(response)
 
     if frame.act == "bye":