play a bit with the code and make it run (at least on my system)

my python is worse, but at least it works now (-: