add draw.py