Update relativity_calculator.py