Update geometry.py

Refinement (fuck it).