Added Python API for workload generator