Added DependencyInjection and Updated WebSocket module