diff --git a/.gradient/workflows/workflow.yaml b/.gradient/workflows/workflow.yaml
index d022d2e..10c05e8 100644
--- a/.gradient/workflows/workflow.yaml
+++ b/.gradient/workflows/workflow.yaml
@@ -1,34 +1,25 @@
-on:
+'on':
   github:
     branches:
       only: main
-
 jobs:
   CloneRepo:
     resources:
-      instance-type: C3
+      instance-type: C5
+    outputs:
+      repo:
+        type: volume
     uses: git-checkout@v1
     with:
-      # url: https://github.com/gradient-ai/mnist-sample.git
       url: context.event.github.url
-      ref: context.event.github.ref
-    outputs:
-      mnist-sample:
-        type: volume
-  TrainModel:
+  HelloWorld:
     resources:
-      instance-type: C3
-    env:
-      PS_MODEL_PATH: /my-trained-model
-    uses: container@v1
-    with:
-      args:
-        - bash
-        - -c
-        - >-
-          cd /inputs/mnist-sample && python mnist.py
-      image: tensorflow/tensorflow:1.15.5-py3
+      instance-type: C5
     needs:
       - CloneRepo
     inputs:
-      mnist-sample: CloneRepo.outputs.mnist-sample
+      repo: CloneRepo.outputs.repo
+    uses: script@v1
+    with:
+      script: python /inputs/repo/hello.py
+      image: tensorflow/tensorflow:1.14.0-py3