Update description

policy/FeudalGainPolicy.py

@@ -21,13 +21,17 @@
 ###############################################################################
 
 
-"""
-Implementation of FeudalGain
-
-Paper: Arxiv reference
+'''
+FeudalGainPolicy.py - What Does The User Want? Information Gain for Hierarchical Dialogue Policy Optimisation
+==================================================
 
 Author: Christian Geishauser
-"""
+
+The implementation of the FeudalGain algorithm that incorporates information gain as intrinsic reward in order to update a Feudal policy.
+The details can be found here: https://arxiv.org/abs/2109.07129
+
+'''
+
 
 import numpy as np
 import random