Now uses np.angle to determine angle