Add solution for problem 54