Added state update via websockets