add import for numpy function