Quick solution for problem 8