use numpy library