For Part 2 of 191-Proj3