Initial implementation of new_axis