Update design_prototype_decent.py

Refactored code.