Update drainS.py