update hw2: add features