This is the first section of the code. I have imported the modules you will need to use. I also defined a few constants for you.

$[1]$:

#general modules

import numpy as np

import matplotlib.pyplot as plt

c $=299792\mathrm{.}0$ #km$/$s

second$\_$hour $=3600\mathrm{.}0$

$[2]$: # Define the time dilation function

def dilation$($speed$)$:

print$($speed$)$

$[3]$: # Define the time requested to travel

def compute$\_$time$($speed$,$ distance$)$:

print$($speed$)$

$[4]$: # Define the function mixing the two previous functions

def total$\_$dilated$($speed$,$ distance$)$:

print$($speed$)$