improve task 3