Completed Assignment 2
Add comments
Add main comments for overall program