[WIP] New axis

Support for creating new axes in arrays, similar to NumPy's newaxis operator.
We should make it work more generally. What is the right way to do this?