Removed unnecessary squeeze()