Update prototype.py

Changed layout.