some changes, but not sure can output right answer