Update create_job_flow.py

This way we can import main() to create a jobflow directly from the code and obtain the jobflow_id.