seems to work
added output for part 3