Updating sample to use WSK