updated output to match updated code