Refactor direct_messages 'connection' -> 'websocket'