Update wsm.cpp