modify Problem4