Switch from `.unwrap()` to `match`

Properly match on message, so client disconnect doesn't trigger a panic

README.md

@@ -14,13 +14,16 @@ fn main () {
         spawn(move || {
             let mut websocket = accept(stream.unwrap()).unwrap();
             loop {
-                let msg = websocket.read_message().unwrap();
+                match websocket.read_message() {
-
+                    Ok(msg) => {
                         // We do not want to send back ping/pong messages.
                         if msg.is_binary() || msg.is_text() {
                             websocket.write_message(msg).unwrap();
                         }
                     }
+                    Err(_) => {}
+                }
+            }
         });
     }
 }