To avoid too many imports, maybe just use ``np.pi`` here (Numpy is already imported in this file)