Updating expected output