updated the client test code