Update geometry_lab.py