Adding updated HW3 code