update draw.py