Add WebSockets goodness. Some code clean.