add task4 answer
add task4 code