This could be better expressed by passing the dtype argument in numpy.arange