Working on prog3 part2