replace np with numpy, remove universal math import