updated assign3 part2