fixed small error from cs155_hw2