Python Distance Formula Physics