Add python solution to challenge 22