Update prototype.py

added bigger while loop.