diff --git a/tf_notification_callback/main.py b/tf_notification_callback/main.py
index 0bdc74f..44df105 100644
--- a/tf_notification_callback/main.py
+++ b/tf_notification_callback/main.py
@@ -62,7 +62,7 @@ def on_train_end(self, logs=None):
     def on_epoch_end(self, epoch, logs=None):
         self.logs_arr.append(logs)
         if not self.getSummary:
-            text = f'*Epoch*: {epoch}\n'
+            text = f'*{self.modelName}*:\n*Epoch*: {epoch}\n'
             for key, value in logs.items():
                 text += f'*{key}*: {value:.2f}\n'
             self.send_message(text, type='text')