Fixed small mistake in norm and swapaxes function.