Create drain.py