added distance calculation but still writing test to confirm it works