Use numpy.linalg.norm to calculate vec norm