Update according to review
Update according to review

Co-Authored-By: Eric Wieser <da64e90baec5d084ce3d8e05f3b022ae6d118ff5@gmail.com>
abalkin/numpy,mattip/numpy,mattip/numpy,pbrod/numpy,simongibbons/numpy,rgommers/numpy,abalkin/numpy,mhvk/numpy,endolith/numpy,madphysicist/numpy,charris/numpy,mattip/numpy,rgommers/numpy,pdebuyl/numpy,mhvk/numpy,seberg/numpy,jakirkham/numpy,anntzer/numpy,simongibbons/numpy,jakirkham/numpy,numpy/numpy,endolith/numpy,charris/numpy,endolith/numpy,rgommers/numpy,mhvk/numpy,WarrenWeckesser/numpy,grlee77/numpy,WarrenWeckesser/numpy,pdebuyl/numpy,pdebuyl/numpy,anntzer/numpy,mhvk/numpy,abalkin/numpy,numpy/numpy,simongibbons/numpy,jakirkham/numpy,madphysicist/numpy,WarrenWeckesser/numpy,rgommers/numpy,seberg/numpy,anntzer/numpy,pbrod/numpy,seberg/numpy,anntzer/numpy,pbrod/numpy,WarrenWeckesser/numpy,charris/numpy,mhvk/numpy,charris/numpy,pbrod/numpy,numpy/numpy,madphysicist/numpy,pbrod/numpy,jakirkham/numpy,grlee77/numpy,madphysicist/numpy,grlee77/numpy,simongibbons/numpy,endolith/numpy,grlee77/numpy,WarrenWeckesser/numpy,seberg/numpy,jakirkham/numpy,simongibbons/numpy,pdebuyl/numpy,numpy/numpy,mattip/numpy,grlee77/numpy,madphysicist/numpy