Add wrapper script for python demo