removed squeeze in favor of slicing to avoid problems with single value arrays