HW 4 - small change