Update 1_pyramid.py