Fixed many minor bugs in UI/Websocket