Simplified new_axis a bit