add new files for hw4 part2