Improved the comments
Added Part 2 code from Homework 0