DZ3 Problem 4.1 done