Add HW4 solution [skip ci]