numpy arrays using arange function