Add test client code