TP task 3 + task 2 fix