I'm trying to implement an equation from a paper in Python (black square equations) -

So far I have a simplified model but I'm unable to generate the intended output (below image); I suspect the issue is with np.exp() though I'm unsure - any suggestions of how I can do this?
import numpy as np
import math
import matplotlib.pyplot as plt
f = 1e6
T = 1/f
Omega = 2*np.pi*f
i = np.arange(0,50e-6,100e-9)
y = np.sin(Omega*i) * (i**2) * np.exp(-i)
plt.figure(1)
plt.plot(i,y,'b-')
plt.grid()
plt.show()

