Add python solution to 001