fix hw11 solution