One line solution for problem 16