Added method to classes, passes initial tests in geo.py script