python solution for problem 24