should have stuck with python code ...