Implemented simple-lattice.py using refactored Geometry