Add SetClient function