Update design.py