Adding comments for HW5 assignment