add py solution to 002