Remove extra math import for pi and replace with np