HW3: minor changes