Add python solution for 002