improve hw2