Updated output to match updated input