fix solution of problem 11