Add a Python solution just for show