Add Python stage for custom python code