Add print lines to match the output