diff --git a/lib/live_flow.js b/lib/live_flow.js
index a910b18..fb0e096 100644
--- a/lib/live_flow.js
+++ b/lib/live_flow.js
@@ -170,7 +170,7 @@ class LiveFlow extends ServerInfo {
                     break;
                 case WS.OP_CONNECT_SUCCESS:
                     if (body[0].code === WS.AUTH_OK) {
-                        log.ok('live_flow -> connect success');
+                        log.ok('live_flow -> connect success' + `(roomid: ${this.roomid})`);
                         msg.type = 'CONNECT_SUCCESS';
                         this.connectState = true;
                         this.heartBeat = this._heartBeat();
diff --git a/lib/message_handle.js b/lib/message_handle.js
index 0a426d9..0289efc 100644
--- a/lib/message_handle.js
+++ b/lib/message_handle.js
@@ -40,9 +40,13 @@ function messageHandle(t) {
             break;
         case "DANMU_MSG":
             console.log(
-                t.info[2][1] +
-                chalk.grey(': ') +
-                t.info[1]
+                (t.info[3].length ? chalk.hex(['#5CC05C', '#5896DE', '#A068F1', '#FF86B2', '#F6BE18', '#AFC1AF', '#B7C9DE', '#D8C9F0', '#FFE5F0', '#F5E8C1'][~~(t.info[3][0] / 4)])('[' + t.info[3][1] + ']') : '')
+                +
+                chalk.hex(['#969696', '#61c05a', '#5896de', '#a068f1', '#ff86b2', '#ff9f3d'][~~(t.info[4][0] / 10)])(t.info[2][1])
+                +
+                chalk.grey(': ')
+                +
+                chalk.hex(t.info[2][7] || '#FFFFFF')(t.info[1])
             );
             break;
         case "USER_TOAST_MSG":
diff --git a/main.js b/main.js
index 8de145b..4c9932b 100644
--- a/main.js
+++ b/main.js
@@ -4,7 +4,7 @@ const { messageHandle } = require("./lib/message_handle");
 (function () {
     const live_flow = new LiveFlow();
     live_flow
-        .setRoomId(80397)
+        .setRoomId(Number(process.argv[2]) || 1)
         .setMessageHandle((msg) => msg.type === 'MESSAGE' && msg.inner.forEach(messageHandle))
         .run();