<x<4 и f(x)=x^2-1, если x>=4. Необходимо построить график этой функции.

import numpy as np
import matplotlib.pyplot as plt
def f(x):
if x < -4:
return -x + 1
else:
return x**2 - 1
x = np.linspace(-5, 5, 100)
y = [f(i) for i in x]
plt.plot(x, y)
plt.xlabel('x')
plt.ylabel('f(x)')
plt.title('График функции у=f(x)')
plt.grid()
plt.show()