Add necessary import to client.py