Update new-client.py

Add remove function
Fixe some bugs