can you add an example about websockets?

I tried this (see below) but can't get the connection event to trigger.

```
'use strict'
const serverlessExpress = require('@codegenie/serverless-express');
const app = require('./app');
const server = serverlessExpress.createServer(app)
const {initWebSocket} = require('./ws');
initWebSocket(server);
exports.handler = (event, context) => {
serverlessExpress.proxy(server, event, context);
};
```

```
const WebSocket = require('ws');

function initWebSocket(server) {

const wss = new WebSocket.Server({server});

wss.on('connection', (ws) => {

console.error('*** connection ****');

ws.on('message', (message) => {
console.error(`Received message: ${message}`);
ws.send(`Echo: ${message}`);
});

ws.on('close', () => {
console.error('user disconnected');
});

ws.on('error', (error) => {
console.error('WebSocket error:', error);
});

ws.send('Welcome to the WebSocket server!');
});

}

module.exports = { initWebSocket };
```