add arange and linspace function