Actually used correct solution for problem 3.