Add python solution 1