HW #4 correction