Cleaning up solution for problem 26!