1 files changed, 4 insertions, 7 deletions
diff --git a/source_file.py b/source_file.py
index 9d10aed..672139c 100644
--- a/source_file.py
+++ b/source_file.py
@@ -28,17 +28,14 @@ def recv_msg(conn):
 	msg = msg.decode(FORMAT)
 	return msg
 
-def terminate_connection(conn):
-	conn.close()
-
-
 msg = "Initializing Client - Python"
 display_msg(msg)
 
 msg = "Initialized Client - Python"
 send_msg(CONN, msg)
 
-msg = recv_msg(CONN)
-display_msg(msg)
+while msg != '!DISCONNECT':
+	msg = recv_msg(CONN)
+	display_msg(msg)
 
-terminate_connection(CONN)
-\ No newline at end of file
+CONN.close()
+\ No newline at end of file