def METHOD_NAME(tau0, nt=100):
    """
    Returns the spatiotemporal temperature profile
    :param float tau0: Time decay parameter.
    """
    r, T0 = getInitialTemperature()
    _, Tf = getFinalTemperature(r)
    # Let the temperature drop until the central temperature is T~100 eV
    #maxt = -tau0 * np.log((100 - Tf[0]) / (T0[0] - Tf[0]))
    maxt = 6e-3
    # Generate time grid
    t = np.linspace(0, maxt, nt).reshape((nt,1))
    # Reshape
    nr = r.size
    T0 = T0.reshape((1,nr))
    Tf = Tf.reshape((1,nr))
    # Calculate temperature evolution
    T = Tf + (T0 - Tf) * np.exp(-t/tau0)
    t = t.reshape((nt,))
    return t, r, T