Update Problem4_code