Add update ws method