Add ws handler