Improved WebSocketControllers scaffolding