Use arange to support numpy ops