Change parameter names and add squeeze()