minor edit on hw2