updated problem4 with 100/100 solution