Add comments for assignment2