Updated WS module