Add setup for the client