Add client test code