HW4 coded added