HW1 p1. minor fix