fixed draw.py