Add python solution to challenge 1