Add missing numpy import in example.