Tinkering with the .py - it seems to work??