Use numpy for range