Handle websocket error gracefully + add retry logic

Handle websocket error gracefully + add retry logic #12