Integrate custom Python code [WIP]