diff --git a/notification_client/client.py b/notification_client/client.py
index 22a8196..de86e33 100644
--- a/notification_client/client.py
+++ b/notification_client/client.py
@@ -32,12 +32,17 @@ class Client:
         reader, writer = await asyncio.open_connection(self.__ip, self.__port)
         await self.handle(reader, writer)
 
+    @staticmethod
+    async def receive_message(reader) -> str:
+        size, = struct.unpack('<L', await reader.readexactly(4))
+        message = await reader.readexactly(size)
+        return message.decode('utf-8')
+
     async def handle(self, reader, writer):
         try:
             while True:
-                size, = struct.unpack('<L', await reader.readexactly(4))
-                message = await reader.readexactly(size)
-                print(f"{message.decode('utf-8') =}")
+                title = await self.receive_message(reader)
+                message = await self.receive_message(reader)
 
         except asyncio.CancelledError:
             print('Something went wrong')
diff --git a/notification_client/main.py b/notification_client/main.py
index f5ed8f1..162199b 100644
--- a/notification_client/main.py
+++ b/notification_client/main.py
@@ -19,7 +19,7 @@
 
 import asyncio
 
-from .client import Client
+from notification_client.client import Client
 
 if __name__ == '__main__':
     client = Client('127.0.0.1', 5555)