Added a python solution to the example