Add python solution to challenge 3