Add python solution to 002